Colorizing Black and White Images with CNN in Python